from scripts.my_sample import load_pipe, infer

load_pipe()

print('模型加载完毕！')
prompt = """
真人照片， 岸边礁石有两个女生，左边女生在唱歌，右边女生弹吉他，两个人面对着镜头，镜头拉远， 可以看到背后是大海，海那边是山，山在画面中间，还可以看到一点点夕阳
""".strip()
width = 1024
height =  1024
num_inference_steps = 32
guidance_scale = 5.0
num_images_per_prompt = 1
seed = 90

#-------------------------------------------------------------------------
#-------------------------------------------------------------------------
#-------------------------------------------------------------------------
import time 
t1 = time.time()
output_dir, images = infer(
    prompt,
    width,
    height,
    num_inference_steps,
    guidance_scale,
    num_images_per_prompt,
    seed,
)
print(f'图片保存在目录：{output_dir}\n')
t2 = time.time()
print(int(t2-t1),'s')